package suanfa;

import java.util.LinkedList;
import java.util.Queue;

public class updateMatrix {
    int m,n;
    int[] dx={1,-1,0,0};
    int[] dy={0,0,1,-1};
    public int[][] updateMatrix(int[][] mat) {
        m=mat.length;
        n=mat[0].length;
        int[][] dist=new int[m][n];
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                dist[i][j]=-1;
            }
        }
        Queue<int[]> q=new LinkedList<>();
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(mat[i][j]==0){
                    dist[i][j]=0;
                    q.add(new int[]{i,j});
                }
            }
        }
        while(!q.isEmpty()){
            int[] t=q.poll();
            int a=t[0],b=t[1];
            for(int i=0;i<4;i++){
                int x=a+dx[i],y=b+dy[i];
                if(x>=0 && y>=0 && x<m && y<n && dist[x][y]==-1){
                    dist[x][y]=dist[a][b]+1;
                    q.add(new int[]{x,y});
                }
            }
        }
        return dist;
    }
}
